from django import forms
from .models import Course ,Lesson


class CreateCourseForm(forms.ModelForm):
    class Meta:
        model = Course
        fields=("title","overview")


class CreateLessonForm(forms.ModelForm):
    class Meta:
        model=Lesson
        fields=['course','title','video','description','attach']

    def __init__(self,user,*args,**kwargs):
        super(CreateLessonForm, self).__init__(*args,**kwargs)
        self.fields['course'].queryset=Course.objects.filter(user=user)